how to use the walrus operator in python